import hou


def setMZnodeColor(kwargs):
    node = kwargs["node"]
    if node.type().name() == 'mz_crowdscontrol':
        node.setColor(hou.Color(1,0,0))
        node.setUserData("nodeshape",'slash')
    elif node.type().name() == 'mz_dopcontrol':
        node.setColor(hou.Color(1,1,0))
        node.setUserData("nodeshape",'bone')
    elif node.type().name() == 'mz_snakecrowdscontrol':
        node.setColor(hou.Color(0,0,0))
        node.setUserData("nodeshape",'pointy')
    elif node.type().name() == 'mz_rendercontrol':
        node.setColor(hou.Color(0.5,0,1))
        node.setUserData("nodeshape",'light')
    elif node.type().name() == 'mz_ui_control':
        node.setColor(hou.Color(0,1,0))
        node.setUserData("nodeshape",'chevron_down')
    elif node.type().name() == 'mz_nodecontrol':
        node.setColor(hou.Color(0.715,1,0.875))
        node.setUserData("nodeshape",'circle')   
    elif node.type().name() == 'mz_nodecontrol':
        node.setColor(hou.Color(0.715,1,0.875))
        node.setUserData("nodeshape",'circle')   
    elif node.type().name() == 'cgai_manager':
        node.setColor(hou.Color(1,0,0))
        node.setUserData("nodeshape",'cgai')   
    elif node.type().name() == 'cgai_manager':
        node.setColor(hou.Color(1,0,0))
        node.setUserData("nodeshape",'cgai')   
    elif node.type().name() == 'cgai_deadline':
        node.setColor(hou.Color(0,1,0))
        node.setUserData("nodeshape",'cgai')   
    elif node.type().name() == 'CGAI_Deadline':
        node.setColor(hou.Color(0,1,0))
        node.setUserData("nodeshape",'cgai')   

setMZnodeColor(kwargs)